#include "Dri_Uart.h"

#define USART_BAUDRATE 9600


#define USART2_PIN_PORT GPIOA
#define USART2_TX_PIN GPIO_Pin_2
#define USART2_RX_PIN GPIO_Pin_3

void Dri_Uart_Init(void)
{
    // 开启时钟USART2
    RCC_APB1PeriphClockCmd(RCC_APB1Periph_USART2, ENABLE);
    // 开启时钟GPIOA
    RCC_AHB1PeriphClockCmd(RCC_AHB1Periph_GPIOA, ENABLE);

    GPIO_InitTypeDef GPIO_InitStructure;
    USART_InitTypeDef USART_InitStructure;
    
    // 配置TX引脚 (PA2)
    GPIO_InitStructure.GPIO_Pin = USART2_TX_PIN;
    GPIO_InitStructure.GPIO_Mode = GPIO_Mode_AF;
    GPIO_InitStructure.GPIO_OType = GPIO_OType_PP;
    GPIO_InitStructure.GPIO_PuPd = GPIO_PuPd_NOPULL;
    GPIO_InitStructure.GPIO_Speed = GPIO_Speed_50MHz;
    GPIO_Init(USART2_PIN_PORT, &GPIO_InitStructure);
    
    // 配置RX引脚 (PA3)
    GPIO_InitStructure.GPIO_Pin = USART2_RX_PIN;
    GPIO_InitStructure.GPIO_Mode = GPIO_Mode_AF;
    GPIO_InitStructure.GPIO_OType = GPIO_OType_PP;
    GPIO_InitStructure.GPIO_PuPd = GPIO_PuPd_NOPULL;
    GPIO_InitStructure.GPIO_Speed = GPIO_Speed_50MHz;
    GPIO_Init(USART2_PIN_PORT, &GPIO_InitStructure);
    
    // 配置GPIO复用功能为USART2
    GPIO_PinAFConfig(USART2_PIN_PORT, GPIO_PinSource2, GPIO_AF_USART2);
    GPIO_PinAFConfig(USART2_PIN_PORT, GPIO_PinSource3, GPIO_AF_USART2);
    
    // 配置USART2参数
    USART_InitStructure.USART_BaudRate = USART_BAUDRATE;
    USART_InitStructure.USART_WordLength = USART_WordLength_8b;
    USART_InitStructure.USART_StopBits = USART_StopBits_1;
    USART_InitStructure.USART_Parity = USART_Parity_No;
    USART_InitStructure.USART_HardwareFlowControl = USART_HardwareFlowControl_None;
    USART_InitStructure.USART_Mode = USART_Mode_Tx | USART_Mode_Rx;
    USART_Init(USART2, &USART_InitStructure);
    
    
    // 使能USART2
    USART_Cmd(USART2, ENABLE);
}

void Dri_Uart_SendByte(uint8_t Byte)
{
    // 等待发送数据寄存器为空
    while (USART_GetFlagStatus(USART2, USART_FLAG_TXE) == RESET);
    
    // 发送数据
    USART_SendData(USART2, Byte);
    
    // 等待发送完成
    while (USART_GetFlagStatus(USART2, USART_FLAG_TC) == RESET);
}

void Dri_Uart_SendArray(uint8_t *Array, uint16_t Length)
{
    uint16_t i;
    for (i = 0; i < Length; i++)
    {
        Dri_Uart_SendByte(Array[i]);
    }
}

void Dri_Uart_SendString(char *String)
{
    uint16_t i;
    for (i = 0; String[i] != '\0'; i++)
    {
        Dri_Uart_SendByte(String[i]);
    }
}

uint8_t Dri_Uart_GetRxFlag(void)
{
    if (USART_GetFlagStatus(USART2, USART_FLAG_RXNE) == SET)
    {
        return 1;
    }
    else
    {
        return 0;
    }
}

uint8_t Dri_Uart_GetRxData(void)
{
    return USART_ReceiveData(USART2);
}

int fputc(int ch, FILE *file)
{
  Dri_Uart_SendByte(ch);
  return ch;
}
